New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Convert histograms to nparray to avoid warning #18

Open

tomhosking wants to merge 1 commit into tingofurro:master from tomhosking:nparraywarning

tomhosking commented Dec 21, 2023

torch throws a warning about creating a tensor from a list of arrays:

summac/model_summac.py:301: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor. (Triggered internally at ../torch/csrc/utils/tensor_new.cpp:261.)
  histograms = torch.FloatTensor(histograms).to(self.device)

This PR converts the list to a single nparray first to avoid this message. I've not noticed a significant change in performance as a result.


          Convert histograms to nparray to avoid warning

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet